/**
 * Copyright (c) 2008, Andrew Carter All rights reserved.
 * 
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions are met:
 * 
 * Redistributions of source code must retain the above copyright notice, this
 * list of conditions and the following disclaimer. Redistributions in binary
 * form must reproduce the above copyright notice, this list of conditions and
 * the following disclaimer in the documentation and/or other materials provided
 * with the distribution. Neither the name of Andrew Carter nor the names of
 * contributors may be used to endorse or promote products derived from this
 * software without specific prior written permission.
 * 
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
 * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
 * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
 * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
 * POSSIBILITY OF SUCH DAMAGE.
 */
package com.acarter.jmejtree.treenode;

import java.awt.Graphics2D;
import java.awt.Image;
import java.awt.RenderingHints;
import java.awt.image.BufferedImage;
import java.io.File;
import java.net.MalformedURLException;
import java.net.URL;
import java.util.HashMap;

import javax.swing.Icon;
import javax.swing.ImageIcon;

import com.acarter.composabletree.ComposableTreeNode;
import com.jme.image.Texture;
import com.jme.scene.state.TextureState;

/**
 * @author Carter
 */
public class TextureTreeNode implements ComposableTreeNode {

	private HashMap<String, ImageIcon> iconCache = new HashMap<String, ImageIcon>();

	/**
	 * @see com.acarter.composabletree.ComposableTreeNode#getChild(java.lang.Object,
	 *      int)
	 */
	public Object getChild(Object parent, int index) {

		if((parent != null) && (parent instanceof TextureState)) {

			TextureState state = (TextureState) parent;

			Texture texture = null;
			int found = -1;

			for(int unit = 0; unit < state.getNumberOfSetTextures(); unit++) {

				texture = state.getTexture(unit);

				if(texture != null)
					found++;

				if(found == index)
					break;
			}

			return texture;
		}
		else
			return null;
	}

	/**
	 * @see com.acarter.composabletree.ComposableTreeNode#getChildCount(java.lang.Object)
	 */
	public int getChildCount(Object parent) {

		int count = 0;

		if((parent != null) && (parent instanceof TextureState)) {

			TextureState state = (TextureState) parent;

			for(int unit = 0; unit < state.getNumberOfSetTextures(); unit++) {

				if(state.getTexture(unit) != null)
					count++;
			}
		}

		return count;
	}

	/**
	 * @see com.acarter.composabletree.ComposableTreeNode#getChildIndex(java.lang.Object,
	 *      java.lang.Object)
	 */
	public int getChildIndex(Object parent, Object child) {

		int index = -1;

		if((parent != null) && (parent instanceof TextureState)) {

			TextureState state = (TextureState) parent;

			for(int unit = 0; unit < state.getNumberOfSetTextures(); unit++) {

				Texture texture = state.getTexture(unit);

				if(texture != null)
					index++;

				if(texture.equals(child))
					break;
			}
		}

		return index;
	}

	/**
	 * @see com.acarter.composabletree.ComposableTreeNode#getParent(java.lang.Object)
	 */
	public Object getParent(Object child) {

		// Textures are not aware of their parent
		return null;
	}

	/**
	 * @see com.acarter.composabletree.ComposableTreeNode#getNodeClassType()
	 */
	public Class<?> getNodeClassType() {

		return Texture.class;
	}

	/**
	 * @see com.acarter.composabletree.ComposableTreeNode#getIcon(Object)
	 */
	public Icon getIcon(Object node) {

		if(node instanceof Texture) {

			String location = ((Texture) node).getImageLocation();

			if(location != null) {

				if(iconCache.containsKey(location))
					return iconCache.get(location);

				try {
					Image image = new ImageIcon(new URL(location)).getImage();

					BufferedImage texture = new BufferedImage(12, 12, BufferedImage.TYPE_INT_RGB);
					Graphics2D graphics2D = texture.createGraphics();
					graphics2D.setRenderingHint(RenderingHints.KEY_INTERPOLATION, RenderingHints.VALUE_INTERPOLATION_BILINEAR);
					graphics2D.drawImage(image, 0, 0, 12, 12, null);

					ImageIcon icon = new ImageIcon(texture);
					iconCache.put(location, icon);
					
					return icon;
				}
				catch (MalformedURLException e) {
					// consume it
				}
			}
		}

		return null;
	}

	/**
	 * @see com.acarter.composabletree.ComposableTreeNode#getNodeText(Object,
	 *      boolean, boolean, boolean, int, boolean)
	 */
	public String getNodeText(Object node, boolean selected, boolean expanded, boolean leaf, int row, boolean hasFocus) {

		if(node instanceof Texture) {

			String location = ((Texture) node).getImageLocation();

			if(location != null) {
				File file = new File(location);

				if(file != null)
					return file.getName();
			}

			String className = node.getClass().getSimpleName();

			if(className.length() == 0)
				className = node.getClass().getName();

			return className;
		}

		return (node != null) ? node.toString() : "";
	}

	/**
	 * @see com.acarter.composabletree.ComposableTreeNode#getNodeToolTipText(Object,
	 *      boolean, boolean, boolean, int, boolean)
	 */
	public String getNodeToolTipText(Object node, boolean selected, boolean expanded, boolean leaf, int row, boolean hasFocus) {

		return (node != null) ? node.toString() : "";
	}

}
